from flask import g
from functools import wraps


# 判断用户是否登录
def login_required(func):
    @wraps(func)
    def wrapper(*args,**kwargs):
        # 在试图函数执行前判断是否登录
        if g.user_id:
            print('user_id>>>', g.user_id)
            return func(*args,**kwargs)
        else:
            return {'message':'请先登录'},401
    return wrapper